package com.loeng.module.system.service.permission;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.ObjUtil;
import com.google.common.collect.Lists;
import com.loeng.framework.common.enums.CommonStatusEnum;
import com.loeng.module.system.controller.admin.permission.vo.menu.MenuListReqVO;
import com.loeng.module.system.dal.dataobject.permission.MenuDO;
import com.loeng.module.system.dal.mysql.permission.MenuMapper;
import com.loeng.module.system.service.tenant.TenantService;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.*;

import static com.loeng.framework.common.util.collection.CollectionUtils.convertList;
import static com.loeng.framework.common.util.collection.CollectionUtils.convertMap;

/**
 * 菜单服务 实现类
 * @author loeng zhuo
 */
@Service
public class MenuServiceImpl implements MenuService {

    @Resource
    private MenuMapper menuMapper;

    @Resource
    private TenantService tenantService;

    /**
     * 获得所有菜单列表
     *
     * @return 菜单列表
     */
    @Override
    public List<MenuDO> getMenuList() {
        return menuMapper.selectList();
    }

    /**
     * 获得菜单列表
     *
     * @param ids 菜单编号数组
     * @return 菜单列表
     */
    @Override
    public List<MenuDO> getMenuList(Collection<Long> ids) {
        // 当 ids 为空时，返回一个空的实例对象
        if (CollUtil.isEmpty(ids)) {
            return Lists.newArrayList();
        }
        return menuMapper.selectBatchIds(ids);
    }

    /**
     * 过滤掉关闭的菜单及其子菜单
     *
     * @param menuList 菜单列表
     * @return 过滤后的菜单列表
     */
    @Override
    public List<MenuDO> filterDisableMenus(List<MenuDO> menuList) {
        if (CollUtil.isEmpty(menuList)){
            return Collections.emptyList();
        }
        // 构建 menuMap，方便通过 id 获取到对应的 menu
        Map<Long, MenuDO> menuMap = convertMap(menuList, MenuDO::getId);

        // 遍历 menu 菜单，查找不是禁用的菜单，添加到 enabledMenus 结果
        List<MenuDO> enabledMenus = new ArrayList<>();
        // 存下递归搜索过被禁用的菜单，防止重复的搜索
        Set<Long> disabledMenuCache = new HashSet<>();
        for (MenuDO menu : menuList) {
            if (isMenuDisabled(menu, menuMap, disabledMenuCache)) {
                continue;
            }
            enabledMenus.add(menu);
        }
        return enabledMenus;
    }

    /**
     * 获得权限对应的菜单编号数组
     *
     * @param permission 权限标识
     * @return 数组
     */
    @Override
    public List<Long> getMenuIdListByPermissionFromCache(String permission) {
        List<MenuDO> menus = menuMapper.selectListByPermission(permission);
        return convertList(menus, MenuDO::getId);
    }

    /**
     * 基于租户，筛选菜单列表
     * 注意，如果是系统租户，返回的还是全菜单
     *
     * @param reqVO 筛选条件请求 VO
     * @return 菜单列表
     */
    @Override
    public List<MenuDO> getMenuListByTenant(MenuListReqVO reqVO) {
        List<MenuDO> menuList = getMenuList();
        // 开启多租户的情况下，需要过滤掉未开通的菜单
        tenantService.handleTenantMenu(menuIds -> menuList.removeIf(menu -> !CollUtil.contains(menuIds, menu.getId())));
        return menuList;
    }

    /**
     * 判断是否禁用菜单
     *
     * @param node              菜单节点
     * @param menuMap           菜单 Map
     * @param disabledMenuCache 禁用的菜单缓存
     * @return 是否禁用
     */
    private boolean isMenuDisabled(MenuDO node, Map<Long, MenuDO> menuMap, Set<Long> disabledMenuCache) {
        // 如果已经判定是禁用的节点，直接结束
        if (disabledMenuCache.contains(node.getId())) {
            return true;
        }

        // 1. 先判断自身是否禁用
        if (CommonStatusEnum.isDisable(node.getStatus())) {
            disabledMenuCache.add(node.getId());
            return true;
        }

        // 2. 遍历到 parentId 为根节点，则无需判断
        Long parentId = node.getParentId();
        if (ObjUtil.equal(parentId, MenuDO.ID_ROOT)) {
            return false;
        }

        // 3. 继续遍历 parent 节点
        MenuDO parent = menuMap.get(parentId);
        if (parent == null || isMenuDisabled(parent, menuMap, disabledMenuCache)) {
            disabledMenuCache.add(node.getId());
            return true;
        }

        return false;
    }

}
